package main

import "go-leetcode/leetcode/utils"

// 437. 路径总和 III
func main() {

}

func pathSum1(root *utils.TreeNode, targetSum int) int { // 2025.02.23 练习, 模拟递归
	ans := 0

	preMap := make(map[int]int)
	preMap[0] = 1

	var dfs func(node *utils.TreeNode, preSum int)
	dfs = func(node *utils.TreeNode, preSum int) {
		if node == nil {
			return
		}
		// 判定是否有前缀和
		preSum += node.Val
		ans += preMap[preSum-targetSum]

		preMap[preSum]++
		dfs(node.Left, preSum)
		dfs(node.Right, preSum)
		preMap[preSum]--
	}
	dfs(root, 0)

	return ans

	//
	//
	//
	//
	//
	//
	//
	//
	//
	//
	//
	//
	//
	//
	//
	//
	return 0
}

func pathSum(root *utils.TreeNode, targetSum int) int {
	preSumMap := map[int]int{0: 1}

	totalSum := 0

	var dfs func(node *utils.TreeNode, preSum int)
	dfs = func(node *utils.TreeNode, preSum int) {
		if node == nil {
			return
		}
		nodeVal := node.Val
		currSum := preSum + nodeVal

		totalSum += preSumMap[currSum-targetSum]

		preSumMap[currSum]++
		dfs(node.Left, currSum)
		dfs(node.Right, currSum)
		preSumMap[currSum]--
	}

	dfs(root, 0)

	return totalSum
}

//给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
//
//路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。
//
//示例 1：
//
//输入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
//输出：3
//解释：和等于 8 的路径有 3 条，如图所示。
//示例 2：
//
//输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
//输出：3
//
//
//提示:
//
//二叉树的节点个数的范围是 [0,1000]
//-109 <= Node.val <= 109
//-1000 <= targetSum <= 1000
